# pip install face_recognition -i https://mirrors.aliyun.com/pypi/simple/
import os
import face_recognition
from tool import insertFace, postInsertFace

path = "D:/项目/python/人脸识别/testimge"
files = os.listdir(path)
testimge_names = []
testimge_faces = []
for file in files:
    # 加载位于指定路径下的图片文件，并将其转换为可供 face_recognition 库使用的图像对象（实际上是 numpy 数组格式）
    image = face_recognition.load_image_file(path + "/" + file)
    face_encodings_list = face_recognition.face_encodings(image)
    if not face_encodings_list:
        print(f"警告: 在文件 {file} 中未检测到人脸.")
        continue  # 或者采取其他行动，如跳过该文件或记录错误日志
    encoding = face_encodings_list[0]  # 默认使用第一张脸的编码
    name = os.path.splitext(os.path.basename(file))[0]
    # insertFace(name, encoding)
    postInsertFace(name, encoding)
print("导入完成")
